from be.model import store
from be.model.store import User, Store_Book, User_Store
class DBConn:
    def __init__(self):
        self.session = store.session
        return
        # self.conn = store.get_db_conn()

    def user_id_exist(self, user_id):
        row = self.session.query(User).filter(User.user_id==user_id).first()
        if row is None:
            return False
        else:
            return True

    def book_id_exist(self, store_id, book_id):
        row = self.session.query(Store_Book).filter(Store_Book.book_id==book_id and Store_Book.store_id==store_id).first()
        if row is None:
            return False
        else:
            return True

    def store_id_exist(self, store_id):
        row = self.session.query(User_Store).filter(User_Store.fk_store_id==store_id).first()
        if row is None:
            return False
        else:
            return True
